From tensorflow/tensorflow

WORKDIR /app
COPY . /app

ENV TF_XLA_FLAGS --tf_xla_cpu_global_jit

RUN apt-get update && apt-get install wget python3-pip python3 -y
RUN wget -P /root/.keras/models https://storage.googleapis.com/tensorflow/keras-applications/resnet/resnet50_weights_tf_dim_ordering_tf_kernels_notop.h5
RUN pip3 install -r /app/requirements.txt -i https://mirrors.aliyun.com/pypi/simple --trusted-host mirrors.aliyun.com

CMD python3 main.py